diff --git a/scann/oss_wrappers/BUILD.bazel b/scann/oss_wrappers/BUILD.bazel
index 834e80b..ac0d048 100644
--- a/scann/oss_wrappers/BUILD.bazel
+++ b/scann/oss_wrappers/BUILD.bazel
@@ -7,8 +7,7 @@ cc_library(
     name = "tf_dependency",
     tags = ["local"],
     deps = [
-        "@local_config_tf//:libtensorflow_framework",
-        "@local_config_tf//:tf_header_lib",
+        "@org_tensorflow//tensorflow/core:tensorflow_opensource",
     ],
 )
 
diff --git a/scann/scann_ops/cc/BUILD.bazel b/scann/scann_ops/cc/BUILD.bazel
index c6ace2c6..46ebfeca 100644
--- a/scann/scann_ops/cc/BUILD.bazel
+++ b/scann/scann_ops/cc/BUILD.bazel
@@ -5,6 +5,15 @@ package(
     licenses = ["notice"],
 )
 
+cc_library(
+    name = "scann_ops_static",
+    alwayslink = 1,
+    deps = [
+        ":scann_ops",
+        ":scann_ops_kernels",
+    ]
+)
+
 cc_binary(
     name = "_scann_ops.so",
     copts = [
@@ -90,17 +99,3 @@ cc_library(
         "@com_google_protobuf//:protobuf",
     ],
 )
-
-pybind_library(
-    name = "scann_npy",
-    srcs = ["scann_npy.cc"],
-    hdrs = ["scann_npy.h"],
-    tags = ["local"],
-    deps = [
-        ":scann",
-        "//scann/base:single_machine_factory_options",
-        "//scann/data_format:dataset",
-        "//scann/utils:io_oss_wrapper",
-        "@com_google_absl//absl/types:span",
-    ],
-)
